#!/usr/bin/env python 
# -*- coding:utf-8 -*-
# 将一个文件夹下图片按比例分在两个文件夹下，比例改0.7这个值即可
import os
import random
#import shutil
from shutil import copy2

def divide_():
    pathName = 'E:\\Python_docs\\Resnet50\\photo'
    for folderName in os.listdir(pathName):
        name = pathName+'\\'+folderName
        trainfiles = os.listdir(name)#（图片文件夹）
        num_train = len(trainfiles)
        #print( "num_train: " + str(num_train) )
        index_list = list(range(num_train))
        #print(index_list)
        random.shuffle(index_list)
        num = 0
        trainDir = 'E:\\Python_docs\\Resnet50\\train\\03\\'#（将图片文件夹中的7份放在这个文件夹下）
        validDir = 'E:\\Python_docs\\Resnet50\\test\\04\\'#（将图片文件夹中的3份放在这个文件夹下）
        for i in index_list:
            fileName = os.path.join(name, trainfiles[i])
            if num < num_train*0.7:
                #print(str(fileName))
                copy2(fileName, trainDir)
            else:
                copy2(fileName, validDir)
            num += 1
        print("划分成功！！！")

divide_()